#!/usr/bin/env python3
# -*- coding: utf-8 -*-

"""
测试新的结构化问答对生成功能
"""

import os
from dotenv import load_dotenv
from data_generator import get_structured_output_for_qa_generation

# 加载环境变量
load_dotenv()


def main():
    """主函数，演示结构化问答对生成功能"""
    print("测试结构化问答对生成功能...")
    
    # 测试文档内容
    test_content = """
    机器学习是人工智能的一个分支，它使计算机能够从数据中学习并做出决策或预测。
    机器学习主要分为三类：监督学习、无监督学习和强化学习。
    监督学习使用标记的数据进行训练，无监督学习在没有标记的数据中发现模式，
    强化学习通过与环境交互来学习最佳行为。
    """
    
    print("文档内容:")
    print(test_content)
    print("\n正在生成问答对...")
    
    try:
        # 生成问答对
        result = get_structured_output_for_qa_generation(
            content=test_content,
            num_pairs=3
        )
        
        # 打印结果
        print("\n生成的问答对:")
        for i, (q, a) in enumerate(zip(result.questions, result.answers), 1):
            print(f"\n问题 {i}: {q}")
            print(f"答案 {i}: {a}")
        
    except Exception as e:
        print(f"调用过程中出现错误: {e}")


if __name__ == "__main__":
    main()